package com.example.callcenter.config;

import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.datatype.jsr310.JavaTimeModule;
import com.fasterxml.jackson.datatype.jsr310.ser.LocalDateTimeSerializer;
import org.springframework.cache.annotation.EnableCaching;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.data.redis.cache.RedisCacheConfiguration;
import org.springframework.data.redis.cache.RedisCacheManager;
import org.springframework.data.redis.connection.RedisConnectionFactory;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.serializer.GenericJackson2JsonRedisSerializer;
import org.springframework.data.redis.serializer.RedisSerializationContext;
import org.springframework.data.redis.serializer.StringRedisSerializer;

import java.time.Duration;
import java.time.LocalDateTime;
import java.time.format.DateTimeFormatter;
import java.util.HashMap;
import java.util.Map;

/**
 * Redis 缓存配置类
 * 配置 RedisTemplate 和 RedisCacheManager，支持 Java 8 日期时间类型的序列化
 */
@Configuration
@EnableCaching  // 开启 Spring Cache 功能
public class RedisConfig {

    /** 日期时间格式化模式 */
    private static final DateTimeFormatter DATE_TIME_FORMATTER = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss");

    /**
     * 配置 RedisTemplate Bean
     * 用于直接操作 Redis，设置键和值的序列化方式
     */
    @Bean
    public RedisTemplate<String, Object> redisTemplate(RedisConnectionFactory factory) {
        RedisTemplate<String, Object> template = new RedisTemplate<>();
        template.setConnectionFactory(factory);

        // 设置键的序列化方式为 String 类型
        StringRedisSerializer stringSerializer = new StringRedisSerializer();
        template.setKeySerializer(stringSerializer);
        template.setHashKeySerializer(stringSerializer);

        // 设置值的序列化方式为 JSON，支持 Java 8 日期时间类型     下面自定义的序列化器
        template.setValueSerializer(createJsonSerializer());
        template.setHashValueSerializer(createJsonSerializer());

        template.afterPropertiesSet();
        return template;
    }

    /**
     * 配置 RedisCacheManager Bean
     * 用于 Spring Cache 抽象，管理不同缓存的配置
     */
    @Bean
    public RedisCacheManager cacheManager(RedisConnectionFactory factory) {
        // 默认缓存配置
        RedisCacheConfiguration defaultConfig = RedisCacheConfiguration.defaultCacheConfig()
                .entryTtl(Duration.ofHours(1))  // 默认缓存过期时间为 1 小时
                .serializeKeysWith(RedisSerializationContext.SerializationPair.fromSerializer(new StringRedisSerializer()))
                //使用下面自定义的序列化器
                .serializeValuesWith(RedisSerializationContext.SerializationPair.fromSerializer(createJsonSerializer()))
                .disableCachingNullValues();  // 禁止缓存 null 值

        // 特定缓存配置（可根据需要扩展）
        Map<String, RedisCacheConfiguration> configs = new HashMap<>();
        configs.put("customerCache", defaultConfig.entryTtl(Duration.ofMinutes(30)));  // 客户缓存 30 分钟过期
        configs.put("taskCache", defaultConfig.entryTtl(Duration.ofMinutes(10)));     // 任务缓存 10 分钟过期

        return RedisCacheManager.builder(factory)
                .cacheDefaults(defaultConfig)
                .withInitialCacheConfigurations(configs)
                .build();
    }

    /**
     * 创建支持 Java 8 日期时间类型的 JSON 序列化器
     * 配置 ObjectMapper 以处理 LocalDateTime 等类型的序列化和反序列化
     */
    private GenericJackson2JsonRedisSerializer createJsonSerializer() {
        ObjectMapper objectMapper = new ObjectMapper();
        JavaTimeModule javaTimeModule = new JavaTimeModule();

        // 注册 LocalDateTime 序列化器，指定日期时间格式
        javaTimeModule.addSerializer(LocalDateTime.class, new LocalDateTimeSerializer(DATE_TIME_FORMATTER));
        objectMapper.registerModule(javaTimeModule);

        return new GenericJackson2JsonRedisSerializer(objectMapper);
    }
}